import os

import openai

os.environ["OPENAI_API_KEY"] = "cc84cb87efef3e9903e8a30abf316b62"  # 将个人token替换到这个位置
os.environ["OPENAI_BASE_URL"] = "https://apitoken.ceba.ceshiren.com/openai/v1/"
os.environ["OPENAI_API_BASE"] = "https://apitoken.ceba.ceshiren.com/openai/v1/"


text_list = """
Python是一种广泛使用的高级编程语言，以其简洁、易读和强大的功能著称。以下是一段200字左右的Python介绍：
Python是一种解释型、高级和通用的编程语言，由Guido van Rossum于20世纪80年代末首次开发，并于1991年首次发布。
Python设计的核心理念是代码的可读性和简洁性。其语法清晰简洁，使得程序员可以用更少的代码表达更多的意思，
这也大大降低了编程的入门门槛。
Python支持多种编程范式，包括面向对象编程、过程化编程和函数式编程。它拥有丰富的标准库和大量的第三方库，
使其在各个领域广泛应用，如Web开发、数据分析、人工智能、科学计算和自动化等。
此外，Python的跨平台特性使得它可以运行在多种操作系统上，包括Windows、Linux和MacOS。
其庞大的社区和丰富的资源也为开发者提供了强大的支持。无论是初学者还是经验丰富的开发者，
Python都是一个极具吸引力的编程语言。
希望这个介绍能帮助你更好地理解Python语言的特点和应用领域。如果你有任何其他问题或需要更详细的信息，请随时告诉我。
"""


while True:
    user_input = input('你的问题:')
    if user_input == '退出':
        break

    messages = [
        {'role': 'system', 'content': '你是一个智能小助手'},
        {'role': 'user', 'content': f'Document:{text_list}'},
        {'role': 'user', 'content': f'Question:{user_input}'},
    ]
    response = openai.chat.completions.create(
        model='gpt-3.5-turbo',
        messages=messages,
        max_tokens=100,
        temperature=0.7
    )
    answer = response.choices[0].message.content
    print(f'答案是:{answer}')
